package com.uniview.controller;

import com.uniview.dto.RespDTO;
import com.uniview.exception.CommonException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;
import java.util.UUID;

/**
 * created by lux
 * 文件处理器
 */
@Slf4j
@RestController
public class FileController {
    /**
     * @desc 文件上传
     */
    @RequestMapping("upload")
    @ResponseBody
    public RespDTO upload (@RequestParam("file") MultipartFile file) {
        // 获取原始名字
        String fileName = file.getOriginalFilename();
        // 文件保存路径
        String filePath = "f:/upload/";
        // 文件重命名，防止重复
        fileName = filePath + UUID.randomUUID() + fileName;
        // 文件对象
        File dest = new File(fileName);
        // 判断路径是否存在，如果不存在则创建
        if(!dest.getParentFile().exists()) {
            dest.getParentFile().mkdirs();
        }
        try {
            // 保存到服务器中
            file.transferTo(dest);
            return RespDTO.onSuc("上传成功");
        } catch (Exception e) {
            e.printStackTrace();
        }
        return RespDTO.onSuc("上传失败");
    }

    /**
     * @desc 文件下载
     */
    @RequestMapping("download")
    public void download(HttpServletResponse response) throws Exception {
        FileInputStream fis=null;
        try {
            // 文件地址，真实环境是存放在数据库中的
            File file = new File("D:\\upload\\a.txt");
            // 穿件输入对象
            fis = new FileInputStream(file);
            // 设置相关格式
            response.setContentType("application/force-download");
            // 设置下载后的文件名以及header
            response.addHeader("Content-disposition", "attachment;fileName=" + "a.txt");
            // 创建输出对象
            OutputStream os = response.getOutputStream();
            // 常规操作
            byte[] buf = new byte[1024];
            int len = 0;
            while((len = fis.read(buf)) != -1) {
                os.write(buf, 0, len);
            }
            fis.close();
            log.info("下载成功");
        } catch (IOException e) {
            throw new CommonException("1111","文件下载");
        } finally {
            fis.close();
        }
    }

}
